package com.lzyyj.common;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

/**
 * MD5 工具类
 * 
 * @author lzyyj
 */
public class MD5Util {

	// 用来将字节转换成 16 进制表示的字符
	private static final char hexDigits[] = { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f' };

	private static MessageDigest messageDigest = null;

	/*
	 * 把byte[]数组转换成十六进制字符串表示形式
	 * 
	 * @param tmp 要转换的byte[]
	 * 
	 * @return 十六进制字符串表示形式
	 */
	private static String convertToHexString(byte[] tmp) {
		String s;
	
		int length = tmp.length;
	
		// 用字节表示就是 16 个字节
		char str[] = new char[length * 2]; // 每个字节用 16 进制表示的话，使用两个字符，
		// 所以表示成 16 进制需要 32 个字符
		int k = 0; // 表示转换结果中对应的字符位置
		for (int i = 0; i < length; i++) { // 从第一个字节开始，对 MD5 的每一个字节
			// 转换成 16 进制字符的转换
			byte byte0 = tmp[i]; // 取第 i 个字节
			str[k++] = hexDigits[byte0 >>> 4 & 0xf]; // 取字节中高 4 位的数字转换,
			// >>> 为逻辑右移，将符号位一起右移
			str[k++] = hexDigits[byte0 & 0xf]; // 取字节中低 4 位的数字转换
		}
		s = new String(str); // 换后的结果转换为字符串
		return s;
	}

	static {
		try {
			messageDigest = MessageDigest.getInstance("MD5");
		} catch (NoSuchAlgorithmException nsaex) {
			System.err.println(MD5Util.class.getName() + "初始化失败，MessageDigest不支持MD5。");
			nsaex.printStackTrace();
		}
	}

	/*
	 * 对一段String生成MD5加密信息
	 * 
	 * @param message 要加密的String
	 * 
	 * @return 生成的MD5信息
	 */
	public static String getMD5String(String sourceString) {
		return getMD5String(sourceString, null);
	}

	/*
	 * 对一段String生成MD5加密信息
	 * 
	 * @param message 要加密的String
	 * 
	 * @return 生成的MD5信息
	 */
	public static String getMD5String(String sourceString, String charsetname) {
		String result = null;

		byte[] source = null;
		try {
			source = charsetname==null||charsetname.trim().length()==0?sourceString.getBytes():sourceString.getBytes(charsetname);
		} catch (UnsupportedEncodingException e) {
			source = new byte[] {};
		}
		result = getMD5String(source);

		return result;
	}

	public static String getMD5String(byte[] bytes) {
		return convertToHexString(messageDigest.digest(bytes));
	}

	/**
	 * 获取文件的MD5字符串
	 * 
	 * @param file
	 *            （文件大小不超过2G）
	 * @return
	 * @throws IOException
	 */
	public static String getMD5String(File file) throws IOException {
		FileInputStream in = new FileInputStream(file);
		FileChannel ch = in.getChannel();
		MappedByteBuffer byteBuffer = ch.map(FileChannel.MapMode.READ_ONLY, 0,
				file.length());
		messageDigest.update(byteBuffer);
		in.close();
		return convertToHexString(messageDigest.digest());
	}
	
	/**
	 * 判断是否是一个MD5加密字符串
	 * @param message
	 * @return
	 */
	public static boolean isMD5String(String message){
		boolean result = false;
		
		if (message!=null){
			String testStr = message.trim().toLowerCase();
			if (testStr.length()==16 || testStr.length()==32){
				result = true;
				for(int i=0;i<testStr.length();i++){
					if (!(testStr.charAt(i)>='0'&&testStr.charAt(i)<='9'||testStr.charAt(i)>='a'&&testStr.charAt(i)<='h')){
						result = false;
						break;
					}
				}
			}
		}
		
		return result;
	}

	public static void main(String[] args) {
		System.out.println("对yyj125用MD5加密后：" + getMD5String("yyj125"));
	}

}
